package io.geekidea.boot.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import io.geekidea.boot.auth.util.LoginUtil;
import io.geekidea.boot.common.BaseEntity;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;
import java.time.LocalDateTime;
import java.util.Objects;

@Slf4j
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {


    /**
     * 使用mp做添加操作时候，这个方法执行
     *
     * @param metaObject 元对象
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        //this.setFieldValByName("createdTime", LocalDateTime.now(), metaObject);
        //this.setFieldValByName("updatedTime", LocalDateTime.now(), metaObject);
        if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseEntity) {
            BaseEntity baseEntity = (BaseEntity) metaObject.getOriginalObject();
            LocalDateTime now = LocalDateTime.now();
            // 创建时间为空，则以当前时间为插入时间
            if (Objects.isNull(baseEntity.getCreatedTime())) {
                baseEntity.setCreatedTime(now);
            }
            // 更新时间为空，则以当前时间为更新时间
            if (Objects.isNull(baseEntity.getUpdatedTime())) {
                baseEntity.setUpdatedTime(now);
            }
            // 此处写获取用户及用户名的业务代码
            String userName = null;
            try {
                userName = LoginUtil.getUsername();
                // 当前登录用户不为空，创建人为空，则当前登录用户为创建人
                if (Objects.nonNull(userName) && Objects.isNull(baseEntity.getCreatedBy())) {
                    baseEntity.setCreatedBy(userName);
                }
                // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
                if (Objects.nonNull(userName) && Objects.isNull(baseEntity.getUpdatedBy())) {
                    baseEntity.setUpdatedBy(userName);
                }
            } catch (Exception e) {
                throw new RuntimeException("token为空",e);
            }

        }
    }

    /**
     * 使用mp做修改操作时候，这个方法执行
     *
     * @param metaObject 元对象
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        log.info("start update fill ....");
        //this.strictUpdateFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now()); // 起始版本 3.3.0(推荐)
        // 或者
        //this.strictUpdateFill(metaObject, "updateTime", () -> LocalDateTime.now(), LocalDateTime.class); // 起始版本 3.3.3(推荐)
        this.setFieldValByName("updatedTime", LocalDateTime.now(), metaObject);
        // 此处写获取用户及用户名的业务代码
        //JwtUser user = WebAppContextHelper.getUser();
        // this.setFieldValByName("updatedBy", user.getUsername(), metaObject);

    }
}
